package tests.org.nicta.learn.factorization.Matchbox;

import static org.junit.Assert.*;

import org.junit.Before;
import org.junit.Test;
import org.nicta.learn.factorization.Matchbox.MatchboxMF;

public class MatchboxTest {
	
	MatchboxMF matchbhoxMF;	
	@Before public void initialize()throws Exception{
		String datasetType = "100K";
		String ratingsFileLocation = "data/MovieLens/100K/u.data.t";
		String usersFileLocation = "data/MovieLens/100K/u.user";
		String moviesFileLocation = "data/MovieLens/100K/u.item";
		Integer dimension = 5;
		matchbhoxMF = new MatchboxMF(datasetType, ratingsFileLocation, 
							usersFileLocation, moviesFileLocation, dimension);
	}
	
	@Test
	public void testFactorize() throws Exception{
		//matchbhoxMF.check_gradient(); 
		matchbhoxMF.factorizeLbfgs();
		//matchbhoxMF.factorize();
	}

	
}

